import sys,os
import pandas as pd
from pathlib import Path

if len(sys.argv)!=3:
	print("Usage:script_name <target_dir> <file_type>")
	exit()

# 合并一个目录下的所有CSV文件
target_dir = Path(sys.argv[1])
target_type = sys.argv[2]  # 要合并的文件类型,csv或xlsx/xls
ttl = target_type.lower()
flist = os.listdir(target_dir)
df_all = pd.DataFrame()
for fn in flist:
	fno = Path(target_dir/fn)
	ext = ((fno.suffix)[1:]).lower()
	if ttl=='csv':
		df = pd.read_csv(os.path.join(target_dir, fn))
	elif ttl=='xlsx' or ttl=='xls':
		df = pd.read_excel(fno.absolute())
	df_all = pd.concat([df_all, df])


if ttl=='csv':
	df_all.to_csv(f'{target_dir}/all.csv', index=False)
elif ttl=='xlsx' or ttl=='xls':
	df_all.to_excel(f'{target_dir}/all.xlsx', index=False)
